package com.zixuan.sina.utils;

import com.jcraft.jsch.Channel;
import com.jcraft.jsch.*;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.util.UUID;

/**
 * Created by jlm on 2019-09-17 17:44
 */
@Component
public class FtpUtils {

    /**
     * 利用JSch包实现SFTP上传文件
     * @param bytes  文件字节流
     * @param fileName  文件名
     * @throws Exception
     */
    public  void sshSftp( String fileName, byte[] bytes) throws Exception{
        Session session = null;
        Channel channel = null;

        JSch jsch = new JSch();


        if(FtpConstant.PORT <=0){
            //连接服务器，采用默认端口
            session = jsch.getSession(FtpConstant.USER, FtpConstant.IP);
        }else{
            //采用指定的端口连接服务器
            session = jsch.getSession(FtpConstant.USER, FtpConstant.IP ,FtpConstant.PORT);
        }

        //如果服务器连接不上，则抛出异常
        if (session == null) {
            throw new Exception("session is null");
        }

        //设置登陆主机的密码
        session.setPassword(FtpConstant.PASSWORD);//设置密码
        //设置第一次登陆的时候提示，可选值：(ask | yes | no)
        session.setConfig("StrictHostKeyChecking", "no");
        //设置登陆超时时间
        session.connect(30000);


        OutputStream outstream = null;

        try {
            //创建sftp通信通道
            channel = (Channel) session.openChannel("sftp");
            channel.connect(1000);
            ChannelSftp sftp = (ChannelSftp) channel;


            //进入服务器指定的文件夹
            sftp.cd(FtpConstant.FILE_PATH);

            //列出服务器指定的文件列表
//            Vector v = sftp.ls("*");
//            for(int i=0;i<v.size();i++){
//                System.out.println(v.get(i));
//            }

            //以下代码实现从本地上传一个文件到服务器，如果要实现下载，对换以下流就可以了
            String suffix = fileName.substring(fileName.lastIndexOf("."));
             fileName = UUID.randomUUID().toString().replace("-", "")+suffix;
            outstream = sftp.put(fileName);
//             bufferedOutputStream = new BufferedOutputStream(outstream);
//            int index=0;
//            int tmp;
//             inputStream = file.getInputStream();
//             bin = new BufferedInputStream(inputStream);
//            byte [] byteArr=new byte[file.getBytes().length];
//            while ((tmp=bin.read(byteArr))!=-1){
//                byteArr[index++] = (byte) tmp;
//                bufferedOutputStream.write(byteArr,0,tmp);
//            }
            outstream.write(bytes);

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            //关流操作
            if(outstream != null){
                outstream.flush();
//                inputStream.close();
//                bin.close();
//                bufferedOutputStream.flush();
//                bufferedOutputStream.close();
                outstream.close();

            }
            if(session != null){
                session.disconnect();
            }
            if(channel != null){
                channel.disconnect();
            }
        }
    }

}
